#!/usr/bin/env python
"""
初始化管理员账号脚本
"""
import os
import sys
import django

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'deepseek_project.settings')
django.setup()

from deepseek_api.models import User

def create_admin():
    """创建管理员账号"""
    # 检查管理员账号是否已存在
    if User.objects.filter(username='admin').exists():
        print('管理员账号已存在')
        return

    # 创建管理员账号
    admin_user = User(
        username='admin',
        email='admin@example.com',
        is_active=True
    )
    admin_user.set_password('admin123')
    admin_user.save()

    print('管理员账号创建成功！')
    print('用户名: admin')
    print('密码: admin123')
    print('邮箱: admin@example.com')

if __name__ == '__main__':
    create_admin()
